# @Time : 2021/2/14 17:07
# @Author : Li Kunlun
# @Description :

import numpy as np

element = 2 * np.arange(4).reshape((2, 2))
# [[0 2]
#  [4 6]]
print(element)
test_elements = [1, 2, 4, 8]
# 判断element是否在test_elements中，返回一个与elements相同形状的布尔值。
mask = np.isin(element, test_elements)
#
# [[False  True]
#  [ True False]]
print(mask)
# [2 4]
print(element[mask])

# invert=True表示返回的mask取反。
mask2 = np.isin(element, test_elements, invert=True)
# [[ True False]
#  [False  True]]
print(mask2)
# [0 6]
print(element[mask2])
